package com.daqsoft.service;

import com.daqsoft.data.Message;
import com.daqsoft.socket.UserWebSocket;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 通知服务层
 * @author 周麟东 on 10:05
 */
@Service
public class NoticeService {
    @Autowired
    private UserWebSocket webSocket;
    /**
     * 消息集合 key为收件人ID value为消息列表
     */
    private static Map<Integer , List<Message>> messages = new HashMap<>();

    /**
     * 向用户发送消息  如果用户不在线则将消息保存到内存中
     */
    public void sendMessage(Message message){
        boolean isOnline = webSocket.sendMessage(message.toJson().toJSONString(),message.getReceiver().getId());
        if (isOnline){
            return;
        }
        List<Message> messageList = messages.get(message.getReceiver().getId());
        if (messageList == null){
            messageList = new ArrayList<>();
        }
        messageList.add(message);
        messages.put(message.getReceiver().getId(),messageList);
    }

    /**
     * 获取用户的消息
     * @param userId 用户ID
     * @return 消息列表
     */
    public List<Message> getFromUserId(int userId){
        return messages.get(userId);
    }

    /**
     * 删除用户消息
     * @param userId 用户ID
     */
    public void removeMessage(int userId){
        messages.remove(userId);
    }

}
